#!/bin/bash

# SPDX-FileCopyrightText: Copyright (c) 1993-2024 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
# SPDX-License-Identifier: Apache-2.0
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

set -e
set -x
rm -rf *.log *.onnx *.nsys-rep *.qdrep *.qdrep-nsys *.trt
#clear

export MODEL_TRAINED=$TRT_COOKBOOK_PATH/00-Data/model/model-trained.onnx

nsys profile \
    --force-overwrite=true \
    -o BuildAndRun \
    trtexec \
        --onnx=$MODEL_TRAINED \
        --saveEngine=model-trained.trt \
    > result-01.log 2>&1

nsys profile \
    --force-overwrite=true \
    -o LoadAndRun \
    trtexec \
        --loadEngine=model-trained.trt \
    > result-02.log 2>&1

echo "Finish"
